Tratamento dos dados
In [ ]:
Copied!
import numpy as np
import os
import zipfile
import cv2
import tensorflow as tf
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import warnings
import numpy as np
import os
import zipfile
import cv2
import tensorflow as tf
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import warnings
Armazendo todos os arquivos em uma lista¶
In [7]:
Copied!
diretorio = 'data/terra'
arquivos = [os.path.join(diretorio, f) for f in sorted(os.listdir(diretorio))]
print(arquivos)
diretorio = 'data/terra'
arquivos = [os.path.join(diretorio, f) for f in sorted(os.listdir(diretorio))]
print(arquivos)
['data/terra/-0.625°N, 102°E.jpg', 'data/terra/-1.75°N, 309.375°E.jpg', 'data/terra/-2°N, 305.125°E.jpg', 'data/terra/-2°N, 306.625°E.jpg', 'data/terra/-3.625°N, 15°E.jpg', 'data/terra/-3°N, 299.75°E.jpg', 'data/terra/-4.125°N, 286.75°E.jpg', 'data/terra/-5.125°N, 298.75°E.jpg', 'data/terra/-8.125°N, 295.875°E.jpg', 'data/terra/0.125°N, 113.5°E.jpg', 'data/terra/0.25°N, 109.375°E.jpg', 'data/terra/15°N, 270°E.jpg', 'data/terra/20.75°N, 283.625°E.jpg', 'data/terra/25°N, 281.375°E.jpg', 'data/terra/28.375°N, 98.75°E.jpg', 'data/terra/36.27°N, 138.57°E.jpg', 'data/terra/36.5°N, 139.5°E.jpg', 'data/terra/4.875°N, 304.75°E.jpg', 'data/terra/44.25°N, 281°E.jpg', 'data/terra/44.75°N, 273.125°E.jpg', 'data/terra/49.5°N, 139.625°E.jpg', 'data/terra/4°N, 286.75°E.jpg', 'data/terra/53.707°N, 96.848°E.jpg', 'data/terra/5°N, 18.625°E.jpg', 'data/terra/6°N, 359.875°E.jpg', 'data/terra/9.75°N, 38.625°E.jpg', 'data/terra/9°N, 349.25°E.jpg']
Verificando tamanho da imagem¶
In [14]:
Copied!
imagem = cv2.imread('data/terra/-0.625°N, 102°E.jpg')
imagem.shape
imagem = cv2.imread('data/terra/-0.625°N, 102°E.jpg')
imagem.shape
Out[14]:
(527, 1585, 3)
In [9]:
Copied!
largura, altura = 1585, 527
largura, altura = 1585, 527
Out[9]:
(527, 1585, 3)
Passando imagens de treino para cores de cinza¶
In [13]:
Copied!
imagens = []
classes = []
for imagem_caminho in arquivos:
try:
imagem = cv2.imread(imagem_caminho)
(H, W) = imagem.shape[:2]
except:
continue
imagem = cv2.resize(imagem, (largura, altura))
imagem = cv2.cvtColor(imagem, cv2.COLOR_BGR2GRAY)
plt.imshow(imagem)
plt.show()
imagem = imagem.ravel()
imagens.append(imagem)
classes.append('terra')
imagens = []
classes = []
for imagem_caminho in arquivos:
try:
imagem = cv2.imread(imagem_caminho)
(H, W) = imagem.shape[:2]
except:
continue
imagem = cv2.resize(imagem, (largura, altura))
imagem = cv2.cvtColor(imagem, cv2.COLOR_BGR2GRAY)
plt.imshow(imagem)
plt.show()
imagem = imagem.ravel()
imagens.append(imagem)
classes.append('terra')
In [ ]:
Copied!
In [ ]:
Copied!